import torch
import torch.nn as nn


class Generator(nn.Module):
    def __init__(self):
        super(Generator, self).__init__()

        self.l1 = nn.Linear(10, 10)
        self.drop = nn.Dropout(0.5)

    def forward(self, input):
        after_linear = self.l1(input)
        out = self.drop(after_linear)
        return out
